embedding ML into flask